from langchain.prompts import load_prompt

# 加载yaml格式的Prompt模板
prompt = load_prompt("../kecheng源码/simple_prompt.yaml")
print(prompt.format(name="小黑", what="恐怖的"))

# 加载json格式的prompt模板
prompt2 = load_prompt("../kecheng源码/simple_prompt.json")
print(prompt2.format_prompt(name="小红", what="搞笑的"))

# 支持加载文件格式的模板，并且对prompt的最终解析结果进行自定义格式化
prompt3 = load_prompt("../kecheng源码/prompt_with_output_parser.json")
print(prompt3.output_parser.parse(
    "George Washington was born in 1732 and died in 1799.\nScore: 1/2"
))
